#include "task/BartClassifier.h"
#include "token/BartTokenizer.h"
#include <vector>
namespace bart {
class BartModel {
private:
  /* data */
  BartTokenizer encoder;
  std::shared_ptr<BartClassifier> classifier;

public:
  BartModel(const std::string &model_path, const std::string &encode_path);
  at::Tensor getPossibility(const std::string &text,
                            const std::string &text_pair);
  ~BartModel() = default;
};
} // namespace bart
